home *** CD-ROM | disk | FTP | other *** search
/ Super PC 33 / Super PC 33 (Shareware).iso / spc / sonido / timidity / source / bag.c next >
Encoding:
C/C++ Source or Header  |  1995-08-03  |  4.0 KB  |  171 lines
  1. /*
  2.  
  3.     bag -- A quick hack to buffer audio data between TiMidity and an
  4.            external player program.
  5.  
  6.     This program is in the Public Domain. It is distributed without
  7.     much hope that it will be useful, and WITHOUT ANY WARRANTY;
  8.     definitely without any implied warranty of MERCHANTABILITY or
  9.     FITNESS FOR ANY PURPOSE.
  10.  
  11.     bag may enable you to play complex files at higher output rates,
  12.     but it's a bit of a pain to use.
  13.  
  14.     Here's a usage example for Linux. splay is a program that (with
  15.     this invocation) plays audio data from stdin. Your flag
  16.     requirements may vary.
  17.  
  18.         timidity -id -Or1lsS -o- -s22 -p48 filenames |
  19.         bag -b 2M -t 1M -c 4k |
  20.         splay -b16 -s22 -S
  21.  
  22. */
  23.  
  24. #include <stdio.h>
  25. #include <stdlib.h>
  26. #include <limits.h>
  27. #include <sys/time.h>
  28. #include <sys/types.h>
  29. #include <unistd.h>
  30.  
  31. static int bytes(char *s)
  32. {
  33.   int i;
  34.   i=atoi(s);
  35.   while (*s>='0' && *s<='9')
  36.     s++;
  37.  
  38.   switch(*s)
  39.     {
  40.     case 'm':
  41.     case 'M':
  42.       i *= 1024;
  43.     case 'k':
  44.     case 'K':
  45.       i *= 1024;
  46.     }
  47.   return i;
  48. }
  49.  
  50. int main(int argc, char **argv)
  51. {
  52.   int bs=1024*1024, rs=4*1024, ws=4*1024, cs=4*1024, thresh=64*1024;
  53.   int c, junk=0, verbosity=0, flush_and_go=0;
  54.   char *bag, *head, *tail;
  55.  
  56.   fd_set readfds, writefds, brokefds;
  57.   struct timeval tmo;
  58.  
  59.   while ((c=getopt(argc, argv, "b:t:r:w:c:hv"))>0)
  60.     switch(c)
  61.       {
  62.       case 'b': bs=bytes(optarg); break;
  63.       case 't': thresh=bytes(optarg); break;
  64.       case 'r': rs=bytes(optarg); break;
  65.       case 'w': ws=bytes(optarg); break;
  66.       case 'c': rs=ws=bytes(optarg); break;
  67.       case 'v': verbosity++; break;
  68.       case '?':
  69.       case 'h':
  70.     printf("\nBag usage:\n\n"
  71.            "  timidity -id -Or[format options] -o- [options and filenames] |\n"
  72.            "  bag [bag options] | audio_player [player options]\n\n"
  73.            "Bag options:\n\n"
  74.            "  -b size    Set buffer size (default %d)\n"
  75.            "  -r size    Set read chunk size (default %d)\n"
  76.            "  -w size    Set write chunk size (default %d)\n"
  77.            "  -c size    Set read/write chunk size\n"
  78.            "  -t size    Set start threshold for writing (default %d)\n"
  79.            "  -v         Verbose mode\n"
  80.            "  -h         This help screen\n\n"
  81.            ,bs,rs,ws,thresh);
  82.     return 0;
  83.       }
  84.  
  85.   if (ws>rs) cs=ws;
  86.   else cs=rs;
  87.   if (bs < ws+rs)
  88.     {
  89.       bs=ws+rs;
  90.       fprintf(stderr, "bag: Buffer size increased to %d bytes\n", bs);
  91.     }
  92.   if (thresh > (bs-rs))
  93.     {
  94.       thresh=bs-rs;
  95.       fprintf(stderr, "bag: Threshold lowered to %d bytes\n", bs);
  96.     }
  97.  
  98.   head=tail=bag=malloc(bs);
  99.   junk=0;
  100.   while (!flush_and_go)
  101.     {
  102.       FD_ZERO(&readfds);
  103.       FD_ZERO(&writefds);
  104.       FD_ZERO(&brokefds);
  105.  
  106.       if (junk<(bs-rs))
  107.     FD_SET(STDIN_FILENO, &readfds);
  108.  
  109.       if (junk>ws && junk>thresh)
  110.     {
  111.       FD_SET(STDOUT_FILENO, &writefds);
  112.       thresh=0;
  113.     }
  114.       
  115.       FD_SET(STDIN_FILENO, &brokefds);
  116.       FD_SET(STDOUT_FILENO, &brokefds);
  117.       
  118.       tmo.tv_sec=1;
  119.       tmo.tv_usec=0;
  120.       if ((c=select(NR_OPEN, &readfds, &writefds, &brokefds, &tmo))<0)
  121.     perror("\nbag: select");
  122.  
  123.       if (verbosity)
  124.     fprintf(stderr, "\rbag: %8d bytes", junk);
  125.  
  126.       if (FD_ISSET(STDOUT_FILENO, &brokefds))
  127.     {
  128.       fprintf(stderr, "\nbag: Elvis has left the building\n");
  129.       return 1;
  130.     }
  131.  
  132.       if (FD_ISSET(STDIN_FILENO, &readfds))
  133.     {
  134.       if ((c=read(STDIN_FILENO, head, 
  135.               (head+rs >= bag+bs) ? (bag+bs-head) : rs))<=0)
  136.         {
  137.           if (c==0 || errno!=EAGAIN)
  138.         flush_and_go=1;
  139.         }
  140.       else
  141.         {
  142.           junk+=c;
  143.           head+=c;
  144.           if (head>=bag+bs)
  145.         head-=bs;
  146.         }
  147.     }
  148.       else if (FD_ISSET(STDIN_FILENO, &brokefds))
  149.     flush_and_go=1;
  150.  
  151.       if (flush_and_go || FD_ISSET(STDOUT_FILENO, &writefds))
  152.     do
  153.       {
  154.         c=(tail+ws >= bag+bs) ? (bag+bs-tail) : ws;
  155.         if (c>junk) c=junk;
  156.         
  157.         if (flush_and_go && verbosity)
  158.           fprintf(stderr, "\rbag: Flushing, %8d bytes", junk);
  159.         
  160.         c=write(STDOUT_FILENO, tail, c);
  161.         junk -= c;
  162.         tail += c;
  163.         if (tail>=bag+bs)
  164.           tail-=bs;
  165.       } while (flush_and_go && junk);
  166.     }
  167.   if (verbosity)
  168.     fprintf(stderr, "\nbag: All done\n");
  169.   return 0;
  170. }
  171.